from torchvision import transforms
import cv2
img_cv = cv2.imread("dataset/train/ants/0013035.jpg")

print(type(img_cv))

img_tenor_trans = transforms.ToTensor()
img_ten = img_tenor_trans(img_cv)
print(type(img_ten))

